angarg12 / nucleogenesis

  function (state, data, visibility, util, format, reaction) {

    let ct = this;

    ct.state = state;

    ct.data = data;

    ct.util = util;

    ct.format = format;

    ct.reaction = reaction;

    ct.adjustAmount = [1, 10, 25, 100];

    ct.update = function(player) {

      processRedox(player);

      processElectronegativity(player);

    };

    function processRedox(player){

      for(let slot of player.element_slots){

        if(!slot){

          continue;

        }

        for (let redox of slot.redoxes) {

          if (!redox.resource || !redox.active || redox.from === redox.to) {

            continue;

          }

          let reactant = ct.generateName(redox.element, redox.from);

          let power = ct.redoxPower(player);

          let number = Math.min(power, player.resources[reactant].number);

          let react = ct.redoxReaction(redox);

          ct.reaction.react(number, react, player);

        }

      }

    }

    function processElectronegativity(player){

      for(let key in data.elements){

        let element = data.elements[key];

        if(element.electronegativity === 0){

          continue;

        }

        let ions = element.anions.concat(element.cations);

		    ions.push(element.main);

        for(let resource of ions){

          if(player.resources[resource].number === 0){

            continue;

          }

          let charge = data.resources[resource].charge || 0;

          let probabilities = probabilityDistribution(key, charge);

          for(let probKey in probabilities){

            if(probKey === charge){

              continue;

            }

            let production = Math.floor(probabilities[probKey]*player.resources[resource].number);

            if(production === 0){

              continue;

            }

            let react = ct.redoxReaction({

              element: key,

              from: charge,

              to: parseInt(probKey, 10)

            });

            // electronegativity is 'for free'

      			react.reactant.eV = 0;

            // FIXME: starvation should fix this

            if(react.reactant['e-']){

              production = Math.min(production, player.resources['e-'].number);

            }

            ct.reaction.react(production, react, player);

          }

        }

      }

    }

    function probabilityDistribution(element, charge){

    	let prob = {};

    	let start = -data.elements[element].electron_affinity.length;

    	let end = charge;

    	// lower than index, affected by negativity

    	rangeProbability(element, prob, start, end, 1, data.elements[element].negative_factor);

    	prob[charge] = 1;

    	start = charge+1;

    	end = data.elements[element].ionization_energy.length+1;

    	// lower than index, affected by positivity

    	rangeProbability(element, prob, start, end, -1, data.elements[element].positive_factor);

    	let sum = 0;

    	for(let i in prob){

    		sum += prob[i];

    	}

    	for(let i in prob){

    		prob[i] /= sum;

    	}

    	return prob;

    }

    function rangeProbability(element, prob, start, end, offset, factor){

    	for(let i = start; i < end; i++){

    		let difference = data.redox[element][i]-data.redox[element][i+offset];

    		if(difference <= 0){

    			difference = -difference;

    		}else{

    			difference = 1/difference;

    		}

    		prob[i] = Math.pow(data.constants.ELECTRONEGATIVITY_CHANCE,Math.abs(i))*factor*difference;

    	}

    }

    /* Calculates the redox power based on the redox upgrades */

    ct.redoxPower = function(player) {

      let level = player.global_upgrades_current.redox_bandwidth;

      let upgrade = data.global_upgrades.redox_bandwidth;

      let basePower = upgrade.power;

      let polynomial = upgrade.power_poly;

      return basePower * Math.floor(Math.pow(level, polynomial));

    };

    /* Writes a redox in the form of a reaction so that we can use the reaction

    service to process it */

    ct.redoxReaction = function (redox) {

      let reactant = ct.generateName(redox.element, redox.from);

      let product = ct.generateName(redox.element, redox.to);

      let energy = redoxEnergy(redox.from, redox.to, redox.element);

      let react = {

        'reactant': {},

        'product': {}

      };

      react.reactant[reactant] = 1;

      react.product[product] = 1;

      if (energy > 0) {

        react.reactant.eV = energy;

      } else if (energy < 0) {

        react.product.eV = -energy;

      }

      let electron = redox.from - redox.to;

      if (electron > 0) {

        react.reactant['e-'] = electron;

      } else if (electron < 0) {

        react.product['e-'] = -electron;

      }

      return react;

    };

    /* Calculates how much energy it takes to go from a redox level to another

    for a given element */

    function redoxEnergy(from, to, element) {

      let energyFrom = data.redox[element][from];

      let energyTo = data.redox[element][to];

      let energy = energyTo - energyFrom;

      return energy;

    }

    /* Generates the name of a ion, e.g. O3+ */

    ct.generateName = function (element, i) {

      if (i === 0) {

        return data.elements[element].main;

      }

      let postfix = '';

      if (Math.abs(i) > 1) {

        postfix = Math.abs(i);

      }

      postfix += getSign(i);

      let name = element + postfix;

      return name;

    };

    function getSign(number) {

      return number > 0 ? '+' : '-';

    }

    /* Calculates the number of redox slots based on the redox upgrades */

    ct.redoxSlots = function (player) {

      let level = player.global_upgrades.redox_slots;

      let upgrade = data.global_upgrades.redox_slots;

      let basePower = upgrade.power;

      let multiplier = upgrade.power_mult;

      return basePower * Math.floor(multiplier * level);

    };

    ct.redoxSize = function (player) {

      let size = 0;

      for(let slot of player.element_slots){

        if(!slot){

          continue;

        }

        size += slot.redoxes.length;

      }

      return size;

    };

    /* Adds a new redox to the player list */

    ct.addRedox = function (player, slot) {

      if(ct.redoxSize(player) >= ct.redoxSlots(player)){

        return;

      }

      slot.redoxes.push({

        resource: data.elements[slot.element].main,

        active: false,

        element: slot.element,

        from: 0,

        to: 1

      });

    };

    ct.removeRedox = function (slot, index) {

      slot.redoxes.splice(index, 1);

    };

    ct.visibleRedox = function(slot) {

      return slot.redoxes;

    };

    ct.adjustLevel = function(player, upgrade, amount){

      player.global_upgrades_current[upgrade] += amount;

      // We cap it between 1 and the current max level

      player.global_upgrades_current[upgrade] = Math.max(1, Math.min(player.global_upgrades_current[upgrade], player.global_upgrades[upgrade]));

    };

    state.registerUpdate('redox', ct.update);

  }

]);